package com.lu.barcode;

import android.graphics.Bitmap;
import android.graphics.Canvas;
import android.graphics.Color;
import android.text.TextUtils;

import com.google.zxing.BarcodeFormat;
import com.google.zxing.EncodeHintType;
import com.google.zxing.WriterException;
import com.google.zxing.common.BitMatrix;
import com.google.zxing.qrcode.QRCodeWriter;

import java.util.Hashtable;

/**
 * Author: luqihua
 * Time: 2017/7/11
 * Description: 二维码生成工具类
 */
public class BarcodeGenerator {

    public static Bitmap createQRBitmap(String content, int size, int Padding) {
        return createQRCodeBitmap(content, size, size,
                "UTF-8", "H", Padding,
                Color.BLACK, Color.WHITE);
    }

    /**
     * 生成简单二维码
     *
     * @param content              字符串内容
     * @param width                二维码宽度
     * @param height               二维码高度
     * @param characterSet         编码方式（一般使用UTF-8）
     * @param errorCorrectionLevel 容错率 L：7% M：15% Q：25% H：35%
     * @param padding              空白边距（二维码与边框的空白区域）
     * @param blockColor           黑色色块
     * @param emptyColor           白色色块
     * @return bitmap 生成的二维码图片
     */
    public static Bitmap createQRCodeBitmap(String content, int width, int height,
                                            String characterSet, String errorCorrectionLevel,
                                            int padding, int blockColor, int emptyColor) {
        // 字符串内容判空
        if (TextUtils.isEmpty(content)) {
            return null;
        }
        // 宽和高>=0
        if (width < 0 || height < 0) {
            return null;
        }
        try {
            /* 1.设置二维码相关配置 */
            Hashtable<EncodeHintType, String> hints = new Hashtable<>();
            // 字符转码格式设置
            if (!TextUtils.isEmpty(characterSet)) {
                hints.put(EncodeHintType.CHARACTER_SET, characterSet);
            }
            // 容错率设置
            if (!TextUtils.isEmpty(errorCorrectionLevel)) {
                hints.put(EncodeHintType.ERROR_CORRECTION, errorCorrectionLevel);
            }
            // 空白边距设置
            hints.put(EncodeHintType.MARGIN, Integer.toString(padding));
            /* 2.将配置参数传入到QRCodeWriter的encode方法生成BitMatrix(位矩阵)对象 */
            BitMatrix bitMatrix = new QRCodeWriter().encode(content, BarcodeFormat.QR_CODE, width, height, hints);
            /* 3.创建像素数组,并根据BitMatrix(位矩阵)对象为数组元素赋颜色值 */
            int realWidth = bitMatrix.getWidth();
            int realHeight = bitMatrix.getHeight();
            int[] pixels = new int[realWidth * realHeight];
            for (int y = 0; y < realHeight; y++) {
                for (int x = 0; x < realWidth; x++) {
                    //bitMatrix.get(x,y)方法返回true是黑色色块，false是白色色块
                    if (bitMatrix.get(x, y)) {
                        pixels[y * realWidth + x] = blockColor;//黑色色块像素设置
                    } else {
                        pixels[y * realWidth + x] = emptyColor;// 白色色块像素设置
                    }
                }
            }
            /*4.创建Bitmap对象,根据像素数组设置Bitmap每个像素点的颜色值,并返回Bitmap对象 */
            return Bitmap.createBitmap(pixels, realWidth, realHeight, Bitmap.Config.ARGB_8888);
        } catch (WriterException e) {
            e.printStackTrace();
            return null;
        }
    }


    /**
     * 向二维码中间添加logo图片(图片合成)
     *
     * @param srcBitmap   原图片（生成的简单二维码图片）
     * @param logoBitmap  logo图片
     * @param logoPercent 百分比 (用于调整logo图片在原图片中的显示大小, 取值范围[0,1] )
     * @return bitmap 生成的二维码图片
     */
    public static Bitmap addLogo(Bitmap srcBitmap, Bitmap logoBitmap, float logoPercent) {
        if (srcBitmap == null) {
            return null;
        }
        if (logoBitmap == null) {
            return srcBitmap;
        }
        //传值不合法时使用0.2F
        if (logoPercent < 0F || logoPercent > 1F) {
            logoPercent = 0.2F;
        }

        /* 1. 获取原图片和Logo图片各自的宽、高值 */
        int srcWidth = srcBitmap.getWidth();
        int srcHeight = srcBitmap.getHeight();
        int logoWidth = logoBitmap.getWidth();
        int logoHeight = logoBitmap.getHeight();

        /* 2. 计算画布缩放的宽高比 */
        float scaleWidth = srcWidth * logoPercent / logoWidth;
        float scaleHeight = srcHeight * logoPercent / logoHeight;

        /* 3. 使用Canvas绘制,合成图片 */
        Bitmap bitmap = Bitmap.createBitmap(srcWidth, srcHeight, Bitmap.Config.ARGB_8888);
        Canvas canvas = new Canvas(bitmap);
        canvas.drawBitmap(srcBitmap, 0, 0, null);

        float pivotX = srcWidth / 2f;
        float pivotY = srcHeight / 2f;

        canvas.scale(scaleWidth, scaleHeight, pivotX, pivotY);
        canvas.drawBitmap(logoBitmap, pivotX - logoWidth / 2f, pivotY - logoHeight / 2f, null);

        return bitmap;
    }
}
